import django_filters

from apps.lookups.calendars.models import CalendarType


class CalendarTypeFilter(django_filters.rest_framework.FilterSet):
    calendar_type = django_filters.CharFilter(field_name='calendar_type', lookup_expr='exact')
    calendar_name = django_filters.CharFilter(field_name='calendar_name', lookup_expr='contains')

    class Meta:
        model = CalendarType
        fields = ['calendar_type', 'calendar_name']
